package com.yht.weblog.common.domain.mapper;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.yht.weblog.common.domain.dos.ArticleDO;
import com.yht.weblog.common.domain.dos.ArticlePublishCountDO;
import org.apache.ibatis.annotations.Select;

import java.time.LocalDate;
import java.util.List;
import java.util.Objects;

/**
 * @author: yht
 * @date: 2024/1/2 22:30
 * @description:
 **/
public interface ArticleMapper extends BaseMapper<ArticleDO> {


    /**
     * 选择分页列表
     *
     * @param current   当前页码
     * @param size      每页数量
     * @param title     文章标题
     * @param startDate 开始日期
     * @param endDate   结束日期
     * @return 分页结果
     */
    default Page<ArticleDO> selectPageList(Long current, Long size, String title, LocalDate startDate, LocalDate endDate) {
        Page<ArticleDO> page = new Page<>(current, size);

        LambdaQueryWrapper<ArticleDO> wrapper = Wrappers.<ArticleDO>lambdaQuery()
                .like(StringUtils.isNotBlank(title), ArticleDO::getTitle, title) // 标题模糊查询
                .ge(Objects.nonNull(startDate), ArticleDO::getCreateTime, startDate)
                .le(Objects.nonNull(endDate), ArticleDO::getCreateTime, endDate)
                .orderByDesc(ArticleDO::getCreateTime);

        return selectPage(page, wrapper);
    }

    /**
     * 根据文章 ID 列表查询文章列表
     *
     * @param current    当前页码
     * @param size       每页数量
     * @param articleIds 文章 ID 列表
     * @return 分页结果
     */
    default Page<ArticleDO> selectPageListByArticleIds(Long current, Long size, List<Long> articleIds) {
        Page<ArticleDO> page = new Page<>(current, size);

        LambdaQueryWrapper<ArticleDO> wrapper = Wrappers.<ArticleDO>lambdaQuery()
                .in(ArticleDO::getId, articleIds)
                .orderByDesc(ArticleDO::getCreateTime);

        return selectPage(page, wrapper);
    }

    /**
     * 根据文章ID查询前一篇文章
     *
     * @param articleId 文章ID
     * @return 前一篇文章的对象
     */
    default ArticleDO selectPreArticle(Long articleId) {
        return selectOne(Wrappers.<ArticleDO>lambdaQuery()
                .orderByAsc(ArticleDO::getId) // 根据文章 ID 升序排序
                .gt(ArticleDO::getId, articleId) // 文章 ID 大于给定的文章 ID
                .last("limit 1"));
    }


    /**
     * 根据文章ID查询后一篇文章
     *
     * @param articleId 文章ID
     * @return 后一篇文章的对象
     */
    default ArticleDO selectNextArticle(Long articleId) {
        return selectOne(Wrappers.<ArticleDO>lambdaQuery()
                .orderByDesc(ArticleDO::getId) // 根据文章 ID 降序排序
                .lt(ArticleDO::getId, articleId) // 文章 ID 小于给定的文章 ID
                .last("limit 1"));
    }

    /**
     * 增加文章阅读量
     *
     * @param articleId 文章ID
     * @return 影响行数
     */
    default int increaseReadNum(Long articleId) {
        // 执行 sql 语句 update t_article set read_num = read_num + 1 where id = #{articleId}
        return update(null, Wrappers.<ArticleDO>lambdaUpdate()
                .eq(ArticleDO::getId, articleId)
                .setSql("read_num = read_num + 1"));
    }

    /**
     * 查询所有文章阅读量
     *
     * @return 文章阅读量列表
     */
    default List<ArticleDO> selectAllReadNum() {
        // 执行 sql 语句 select read_num from t_article
        return selectList(Wrappers.<ArticleDO>lambdaQuery()
                .select(ArticleDO::getReadNum));
    }

    /**
     * 查询指定日期范围内的文章发布数量
     *
     * @param startDate 开始日期
     * @param endDate   结束日期
     * @return 文章发布数量列表
     */
    @Select("SELECT DATE(create_time) AS date, COUNT(*) AS count\n" +
            "FROM t_article\n" +
            "WHERE create_time >= #{startDate} AND create_time < #{endDate}\n" +
            "GROUP BY DATE(create_time)")
    List<ArticlePublishCountDO> selectDateArticlePublishCount(LocalDate startDate, LocalDate endDate);

}
